
from mcp.server import FastMCP
from core import (
    add_container_tools,
    add_image_tools,
    add_shell_tools,
    add_file_tools,
    add_code_tools,
    add_jimeng_tools,
    add_tencent_tools
)
import docker

_docker: docker.DockerClient = docker.from_env()
mcp_server = FastMCP("docker-server", host='0.0.0.0', port='8920')

add_container_tools(mcp_server, _docker)    # docker容器管理
add_image_tools(mcp_server, _docker)        # docker镜像管理
add_shell_tools(mcp_server, _docker)        # docker-Shell执行
add_file_tools(mcp_server, _docker)         # docker-文件操作
add_code_tools(mcp_server, _docker)         # docker-代码执行
add_jimeng_tools(mcp_server)                # 即梦图片/视频处理
add_tencent_tools(mcp_server)               # 腾讯OCR识别


if __name__ == '__main__':
    # mcp_server.run(transport="stdio")
    mcp_server.run(transport='sse')
